﻿// NNUE評価関数の入力特徴量HalfKP_vmの定義

#include "../../../config.h"

#if defined(EVAL_NNUE)

#include "half_kp_vm.h"
#include "index_list.h"

namespace YaneuraOu {
namespace Eval::NNUE::Features {

// 玉の位置とBonaPieceから特徴量のインデックスを求める
template <Side AssociatedKing>
inline IndexType HalfKP_vm<AssociatedKing>::MakeIndex(Square sq_k, BonaPiece p) {
	if (sq_k >= SQ_61) {
		// 玉が6筋～9筋にいる場合、4筋～1筋に反転する。
		sq_k = Mir(sq_k);

		if (p >= fe_hand_end) {
			// 持駒は反転しない。
			IndexType piece_index = (p - fe_hand_end) / SQ_NB;
			Square sq_p = static_cast<Square>((p - fe_hand_end) % SQ_NB);
			sq_p = Mir(sq_p);
			p = static_cast<BonaPiece>(fe_hand_end + piece_index * static_cast<IndexType>(SQ_NB) + sq_p);
		}
	}
	return static_cast<IndexType>(fe_end) * static_cast<IndexType>(sq_k) + p;
}

// 駒の情報を取得する
template <Side AssociatedKing>
inline void HalfKP_vm<AssociatedKing>::GetPieces(
	const Position& pos, Color perspective,
	BonaPiece** pieces, Square* sq_target_k) {
	*pieces = (perspective == BLACK) ?
		pos.eval_list()->piece_list_fb() :
		pos.eval_list()->piece_list_fw();
	const PieceNumber target = (AssociatedKing == Side::kFriend) ?
		static_cast<PieceNumber>(PIECE_NUMBER_KING + perspective) :
		static_cast<PieceNumber>(PIECE_NUMBER_KING + ~perspective);
	*sq_target_k = static_cast<Square>(((*pieces)[target] - f_king) % SQ_NB);
}

// 特徴量のうち、値が1であるインデックスのリストを取得する
template <Side AssociatedKing>
void HalfKP_vm<AssociatedKing>::AppendActiveIndices(
	const Position& pos, Color perspective, IndexList* active) {
	// コンパイラの警告を回避するため、配列サイズが小さい場合は何もしない
	if (RawFeatures::kMaxActiveDimensions < kMaxActiveDimensions) return;

	BonaPiece* pieces;
	Square sq_target_k;
	GetPieces(pos, perspective, &pieces, &sq_target_k);
	for (PieceNumber i = PIECE_NUMBER_ZERO; i < PIECE_NUMBER_KING; ++i) {
		active->push_back(MakeIndex(sq_target_k, pieces[i]));
	}
}

// 特徴量のうち、一手前から値が変化したインデックスのリストを取得する
template <Side AssociatedKing>
void HalfKP_vm<AssociatedKing>::AppendChangedIndices(
	const Position& pos, Color perspective,
	IndexList* removed, IndexList* added) {
	BonaPiece* pieces;
	Square sq_target_k;
	GetPieces(pos, perspective, &pieces, &sq_target_k);
	const auto& dp = pos.state()->dirtyPiece;
	for (int i = 0; i < dp.dirty_num; ++i) {
		if (dp.pieceNo[i] >= PIECE_NUMBER_KING) continue;
		const auto old_p = static_cast<BonaPiece>(
			dp.changed_piece[i].old_piece.from[perspective]);
		removed->push_back(MakeIndex(sq_target_k, old_p));
		const auto new_p = static_cast<BonaPiece>(
			dp.changed_piece[i].new_piece.from[perspective]);
		added->push_back(MakeIndex(sq_target_k, new_p));
	}
}

template class HalfKP_vm<Side::kFriend>;
template class HalfKP_vm<Side::kEnemy>;

} // namespace Eval::NNUE::Features
} // namespace YaneuraOu

#endif  // defined(EVAL_NNUE)
